手把手带你通过Python调用接口实现抠图并改底色-世界最新

import osimport requestsimport base64import cv2import numpy as npfrom PIL import Imagefrom pathlib import Pathpath = os.getcwd()paths = list(Path(path).glob("*"))

def get_access_token():    url = "https://aip.baidubce.com/oauth/2.0/token"    data = {        "grant_type": "client_credentials",  # 固定值        "client_id": "替换成你的API Key",  # 在开放平台注册后所建应用的API Key        "client_secret": "替换成你的Secret Key"  # 所建应用的Secret Key    }    res = requests.post(url, data=data)    res = res.json()    access_token = res["access_token"]    return access_token

def removebg():    try:        request_url = "https://aip.baidubce.com/rest/2.0/image-classify/v1/body_seg"        # 二进制方式打开图片文件        f = open(name, "rb")        img = base64.b64encode(f.read())        params = {"image":img}        access_token = get_access_token()        request_url = request_url + "?access_token=" + access_token        headers = {"content-type": "application/x-www-form-urlencoded"}        response = requests.post(request_url, data=params, headers=headers)        if response:            res = response.json()["foreground"]            png_name=name.split(".")[0]+".png"            with open(png_name,"wb") as f:                data = base64.b64decode(res)                f.write(data)            fullwhite(png_name) #png图片底色填充，视情况舍去            png_jpg(png_name) #png格式转jpg,视情况舍去            os.remove(png_name) #删除原png图片，视情况舍去            print(name+"\t处理成功！")    except Exception as e:        pass

def fullwhite(png_name):    im = Image.open(png_name)    x,y = im.size    try:        p = Image.new("RGBA", im.size, (255,255,255))        # 使用白色来填充背景,视情况更改        p.paste(im, (0, 0, x, y), im)        p.save(png_name)    except:        pass

#compress_rate：数值越小照片越模糊def resize(compress_rate = 0.5):    im = Image.open(name)    w, h = im.size    im_resize = im.resize((int(w*compress_rate), int(h*compress_rate)))    resize_w, resieze_h = im_resize.size    #quality 代表图片质量，值越低越模糊    im_resize.save(name)    im.close()

def get_size():    size = os.path.getsize(name)    return size / 1024

def png_jpg(png_name):    im = Image.open(png_name)    bg=Image.new("RGB",im.size,(255,255,255))    bg.paste(im)    jpg_name = png_name.split(".")[0]+".jpg"    #quality 代表图片质量，值越低越模糊    bg.save(jpg_name,quality=70)    im.close()

if __name__ == "__main__":    for i in paths:        name = os.path.basename(i.name)        if(name==os.path.basename(__file__)):            continue        size = get_size()        ##照片压缩        while size >=900:            size = get_size()            resize()           removebg()        print(" ")

#人像分割import osimport requestsimport base64import cv2import numpy as npfrom PIL import Imagefrom pathlib import Pathpath = os.getcwd()paths = list(Path(path).glob("*"))def get_access_token():    url = "https://aip.baidubce.com/oauth/2.0/token"    data = {        "grant_type": "client_credentials",  # 固定值        "client_id": "替换成你的API Key",  # 在开放平台注册后所建应用的API Key        "client_secret": "替换成你的Secret Key"  # 所建应用的Secret Key    }    res = requests.post(url, data=data)    res = res.json()    access_token = res["access_token"]    return access_tokendef png_jpg(png_name):    im = Image.open(png_name)    bg=Image.new("RGB",im.size,(255,255,255))    bg.paste(im)    jpg_name = png_name.split(".")[0]+".jpg"    #quality 代表图片质量，值越低越模糊    bg.save(jpg_name,quality=70)    im.close()#compress_rate：数值越小照片越模糊def resize(compress_rate = 0.5):    im = Image.open(name)    w, h = im.size    im_resize = im.resize((int(w*compress_rate), int(h*compress_rate)))    resize_w, resieze_h = im_resize.size    #quality 代表图片质量，值越低越模糊    im_resize.save(name)    im.close()    def get_size():    size = os.path.getsize(name)    return size / 1024    def fullwhite(png_name):    im = Image.open(png_name)    x,y = im.size    try:        # 使用白色来填充背景        # (alpha band as paste mask).        p = Image.new("RGBA", im.size, (255,255,255))        p.paste(im, (0, 0, x, y), im)        p.save(png_name)    except:        passdef removebg():    try:        request_url = "https://aip.baidubce.com/rest/2.0/image-classify/v1/body_seg"        # 二进制方式打开图片文件        f = open(name, "rb")        img = base64.b64encode(f.read())        params = {"image":img}        access_token = get_access_token()        request_url = request_url + "?access_token=" + access_token        headers = {"content-type": "application/x-www-form-urlencoded"}        response = requests.post(request_url, data=params, headers=headers)        if response:            res = response.json()["foreground"]            png_name=name.split(".")[0]+".png"            with open(png_name,"wb") as f:                data = base64.b64decode(res)                f.write(data)            fullwhite(png_name)            png_jpg(png_name)            os.remove(png_name)            print(name+"\t处理成功！")    except Exception as e:        passif __name__ == "__main__":    for i in paths:        name = os.path.basename(i.name)        if(name==os.path.basename(__file__)):            continue        size = get_size()        ##照片压缩        while size >=900:            size = get_size()            resize()           removebg()        print(" ")